import onnx
from onnxsim import simplify

# 加载ONNX模型
model = onnx.load('linear.onnx')

# 剪枝模型权重
model = simplify(model)

# 保存剪枝后的模型
onnx.save(model, 'model_pruned.onnx')
